package com.yc.ssm.util;

import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;

import javax.imageio.ImageIO;


public class WaterMarkUtil {
	private static final String FONT_NAME = "宋体";
	private static final int FONT_STYLE = Font.BOLD;	// 粗体
	private static final int FONT_SIZE = 20; // 字体大小
	private static final Color FONT_COLOR = Color.RED;	// 字体颜色
	private static final String MARK_TEXT = "源辰信息";	// 水印文字
	
	/**
	 * 给指定的图片右下角加上水印文字
	 * 等用户将图片上传到服务器之后再水印
	 * @param pic
	 */
	public static void textWaterMark(File pic) {
		FileOutputStream fos = null;
		try {
			// 1、获取要添加水印的图片
			Image image = ImageIO.read(pic);
			
			// 2、获取图片的大小
			int width = image.getWidth(null);
			int height = image.getHeight(null);
			
			// 3、创建图片缓冲对象
			BufferedImage bufferedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
			
			// 4、获取缓冲区图片的会绘制工具
			Graphics2D g = bufferedImage.createGraphics();
			
			// 5、绘制原图片
			g.drawImage(image, 0, 0, width, height, null);
			
			// 6、再绘制水印
			// (1) 设置水印文字字体、样式
			g.setFont(new Font(FONT_NAME, FONT_STYLE, FONT_SIZE));
			
			// (2) 设置字体颜色
			g.setColor(FONT_COLOR);
			
			// (3) 计算文本大小，以便定位
			int x = width - getTextWidth(MARK_TEXT) * FONT_SIZE;
			int y = height - FONT_SIZE;	// 这个水印文本距顶部的距离为图片的高度-字体大小
			
			// (4) 设置透明度
			g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, 0.8F));
			
			// (5) 设置文本的位置
			g.drawString(MARK_TEXT, x, y);
			
			// 7、释放工具
			g.dispose();
			
			fos = new FileOutputStream(pic.getAbsolutePath());
			
			// 8、将图片缓冲区中的数据写入到文件中
			ImageIO.write(bufferedImage, "JPEG", fos);
			
		} catch (IOException e) {
			e.printStackTrace();
		}finally {
			if (fos != null) {
				try {
					fos.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}

	/**
	 * 获取文本的宽度
	 * @param text
	 * @return
	 */
	private static int getTextWidth(String text) {
		int length = text.length();	// 获取水印字符串的长度
		String temp = null;
		// 循环获取每一个字符
		for(int i = 0, len = text.length(); i < len; ++ i) {
			temp = String.valueOf(text.charAt(i));	// 获取当前循环得到的字符
			if(temp.getBytes().length > 1) {	// 说明这个不是一个英文字符
				++ length;
			}
		}
		length = length % 2 == 0 ? length / 2 : length / 2 + 1;
		return length;
	}
	
	
	public static void ImgWaterMark(File pic) {
		FileOutputStream fos = null;
		try {
			// 1、获取要添加水印的图片
			Image image = ImageIO.read(pic);
			
			// 2、获取图片的大小
			int width = image.getWidth(null);
			int height = image.getHeight(null);
			
			// 3、创建图片缓冲对象
			BufferedImage bufferedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
			
			// 4、获取缓冲区图片的会绘制工具
			Graphics2D g = bufferedImage.createGraphics();
			
			// 5、绘制原图片
			g.drawImage(image, 0, 0, width, height, null);
			
			// 6、再绘制水印
			// (1) 水印图片
			String waterMarkImage = ConstantInfo.basepath + "/snack/images/yc.png";
			Image srcWaterMark = ImageIO.read(new File(waterMarkImage));
			
			// (2) 获取水印图片的宽度和高度
			int widthWaterMark = srcWaterMark.getWidth(null);
			int heightWaterMark = srcWaterMark.getHeight(null);
			
			// (4) 设置透明度
			g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, 0.8F));
			
			// (5) 设置图片的位置
			g.drawImage(srcWaterMark, (width - widthWaterMark), (height - heightWaterMark), widthWaterMark, heightWaterMark, null);
			
			// 7、释放工具
			g.dispose();
			
			fos = new FileOutputStream(pic.getAbsolutePath());
			
			// 8、将图片缓冲区中的数据写入到文件中
			ImageIO.write(bufferedImage, "JPEG", fos);
			
		} catch (IOException e) {
			e.printStackTrace();
		}finally {
			if (fos != null) {
				try {
					fos.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}
}
